package com.baicai.baicaiknowledgeai.controller;

import com.baicai.baicaiknowledgeai.entity.User;
import jakarta.servlet.http.HttpSession;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;

/**
 * 页面跳转控制器
 * @author 白菜
 */
@Controller
public class PageController {

    /**
     * 根路径，直接导向登录页面
     */
    @GetMapping("/")
    public String root() {
        return "redirect:/login.html";
    }

    /**
     * 登录页的路由
     * 如果用户已登录，直接重定向到聊天页面
     */
    @GetMapping("/login")
    public String loginPage(HttpSession session) {
        if (session.getAttribute("user") != null) {
            return "redirect:/chat.html";
        }
        // Spring Boot 会自动在 static 目录下寻找 login.html
        return "redirect:/login.html";
    }

    /**
     * 聊天主页面的路由
     * 如果用户未登录，重定向回登录页
     */
    @GetMapping("/chat")
    public String chatPage(HttpSession session) {
        User user = (User) session.getAttribute("user");
        if (user == null) {
            return "redirect:/login.html";
        }
        // 重定向到静态的 chat.html
        return "redirect:/chat.html";
    }
}